package com.jpro.module.system.dal.dataobject.permission;

import cn.hutool.core.collection.CollUtil;
import com.jpro.framework.common.enums.CommonStatusEnum;
import com.jpro.framework.common.util.json.JsonUtils;
import com.jpro.framework.mybatis.core.type.JsonLongSetTypeHandler;
import com.jpro.framework.tenant.core.db.TenantBaseDO;
import com.jpro.module.system.controller.admin.permission.vo.permission.SiteChannelVO;
import com.jpro.module.system.enums.permission.DataScopeEnum;
import com.jpro.module.system.enums.permission.RoleTypeEnum;
import com.baomidou.mybatisplus.annotation.KeySequence;
import com.baomidou.mybatisplus.annotation.TableField;
import com.baomidou.mybatisplus.annotation.TableId;
import com.baomidou.mybatisplus.annotation.TableName;
import com.baomidou.mybatisplus.extension.handlers.AbstractJsonTypeHandler;
import com.baomidou.mybatisplus.extension.handlers.JacksonTypeHandler;
import lombok.Data;
import lombok.EqualsAndHashCode;

import java.util.Collections;
import java.util.HashSet;
import java.util.List;
import java.util.Set;

/**
 * 角色 DO
 *
 * @author ruoyi
 */
@TableName(value = "system_role", autoResultMap = true)
@KeySequence("system_role_seq") // 用于 Oracle、PostgreSQL、Kingbase、DB2、H2 数据库的主键自增。如果是 MySQL 等数据库，可不写。
@Data
@EqualsAndHashCode(callSuper = true)
public class RoleDO extends TenantBaseDO {

    /**
     * 角色ID
     */
    @TableId
    private Long id;
    /**
     * 角色名称
     */
    private String name;
    /**
     * 角色标识
     *
     * 枚举
     */
    private String code;
    /**
     * 角色排序
     */
    private Integer sort;
    /**
     * 角色状态
     *
     * 枚举 {@link CommonStatusEnum}
     */
    private Integer status;
    /**
     * 角色类型
     *
     * 枚举 {@link RoleTypeEnum}
     */
    private Integer type;
    /**
     * 备注
     */
    private String remark;

    /**
     * 数据范围
     *
     * 枚举 {@link DataScopeEnum}
     */
    private Integer dataScope;
    /**
     * 数据范围(指定部门数组)
     * <p>
     * 适用于 {@link #dataScope} 的值为 {@link DataScopeEnum#DEPT_CUSTOM} 时
     */
    @TableField(typeHandler = JacksonTypeHandler.class)
    private Set<Long> dataScopeDeptIds;

    /**
     * 栏目数据范围（1：全部数据权限 2：自定数据权限）
     * <p>
     * 枚举 {@link DataScopeEnum}
     */
    private Integer channelDataScope;
    /**
     * 栏目管理的栏目数据权限范围
     * <p>
     * 适用于 {@link #dataScope} 的值为 {@link DataScopeEnum#CHANNEL_CUSTOM} 时
     */
    @TableField(typeHandler = SiteChannelTypeHandler.class)
    private List<SiteChannelVO> dataScopeChannelIds;

    /**
     * 文档数据范围（1：全部数据权限 2：自定数据权限）
     * <p>
     * 枚举 {@link DataScopeEnum}
     */
    private Integer docDataScope;
    /**
     * 文档类栏目数据权限范围
     * <p>
     * 适用于 {@link #dataScope} 的值为 {@link DataScopeEnum#CHANNEL_CUSTOM} 时
     */
    @TableField(typeHandler = SiteChannelTypeHandler.class)
    private List<SiteChannelVO> dataScopeDocChannelIds;

    /**
     * 站点数据范围（1：全部数据权限 2：自定数据权限）
     * <p>
     * 枚举 {@link DataScopeEnum}
     */
    private Integer siteDataScope;
    /**
     * 站点数据权限范围
     * <p>
     * 适用于 {@link #dataScope} 的值为 {@link DataScopeEnum#SITE_CUSTOM} 时
     */
    @TableField(typeHandler = JsonLongSetTypeHandler.class)
    private Set<Long> dataScopeSiteIds;

    @TableField(typeHandler = JsonLongSetTypeHandler.class, exist = false)
    private Set<Long> scopeChannelIdList;

    @TableField(typeHandler = JsonLongSetTypeHandler.class, exist = false)
    private Set<Long> scopeDocChannelIdList;

    /**
     * 获取栏目类权限的所有栏目
     *
     * @return
     */
    public Set<Long> getScopeChannelIdList() {
        if (CollUtil.isEmpty(dataScopeChannelIds)) {
            return Collections.emptySet();
        }
        Set<Long> channelIdList = new HashSet<>();
        dataScopeChannelIds.forEach(c -> channelIdList.addAll(c.getChannelIds()));
        return channelIdList;
    }


    /**
     * 获取内容类权限的所有栏目
     *
     * @return
     */
    public Set<Long> getScopeDocChannelIdList() {
        if (CollUtil.isEmpty(dataScopeDocChannelIds)) {
            return Collections.emptySet();
        }
        Set<Long> channelIdList = new HashSet<>();
        dataScopeDocChannelIds.forEach(c -> channelIdList.addAll(c.getChannelIds()));
        return channelIdList;
    }

    public static class SiteChannelTypeHandler extends AbstractJsonTypeHandler<List<SiteChannelVO>> {

        public SiteChannelTypeHandler(Class<?> type) {
            super(type);
        }

        @Override
        public List<SiteChannelVO> parse(String json) {
            return JsonUtils.parseArray(json, SiteChannelVO.class);
        }

        @Override
        public String toJson(List<SiteChannelVO> obj) {
            return JsonUtils.toJsonString(obj);
        }

    }

}
